package user

import (
	"go-iot/app/models"
	"go-iot/app/common/mysql"
)

func GetUserRelRoleByUserId(userId int64) ([]models.UserRelRole, error) {
	var userRoles []models.UserRelRole
	
	// 使用GORM查询用户关联的角色
	err := mysql.DB.Where("user_id = ?", userId).Find(&userRoles).Error
	if err != nil {
		return nil, err
	}
	
	return userRoles, nil
}

func AddUserRelRole(userId, roleId int64) error {
	// 创建用户角色关联对象
	userRole := &models.UserRelRole{
		UserId: userId, 
		RoleId: roleId,
	}
	
	// 使用GORM插入数据
	return mysql.DB.Create(userRole).Error
}

func DeleteUserRelRoleByUserId(userId int64) error {
	// 使用GORM删除用户关联的所有角色
	return mysql.DB.Where("user_id = ?", userId).Delete(&models.UserRelRole{}).Error
}
